package com.xiaoshuidi.cloud.module.rooms.repository;

import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.xiaoshuidi.cloud.module.rooms.mapper.RoomHouseEstimateInfoMapper;
import com.xiaoshuidi.cloud.module.rooms.pojo.RoomHouseEstimateInfo;
import org.springframework.stereotype.Repository;

/**
 * @Author Feng
 * @date 2023/5/29
 */
@Repository
public class RoomHouseEstimateInfoRepository extends ServiceImpl<RoomHouseEstimateInfoMapper, RoomHouseEstimateInfo> {

    /**
     * 根据测算表ID获取
     *
     * @param estimateId
     * @return
     */
    public RoomHouseEstimateInfo getByEstimateId(Long estimateId) {
        LambdaQueryWrapper<RoomHouseEstimateInfo> wrapper = Wrappers.lambdaQuery();
        wrapper.eq(RoomHouseEstimateInfo::getEstimateId, estimateId);
        return getOne(wrapper);
    }
}
